package com.algorithm.quick;
/*
 *@Author:DOWN
 *@Date:2022/1/20 10:51
 *@Comment:给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
           此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

           示例 1：
           输入：nums = [2,0,2,1,1,0]
           输出：[0,0,1,1,2,2]

           示例 2：
           输入：nums = [2,0,1]
           输出：[0,1,2]

           示例 3：
           输入：nums = [0]
           输出：[0]

           示例 4：
           输入：nums = [1]
           输出：[1]
 */

import java.util.Arrays;
import java.util.Random;

public class Solution75 {
    public static void sortColors(int[] nums) {
        sortThreeWays(nums,0,nums.length-1);
    }
    public static  void sortThreeWays(int[] arrs, int l, int r) {
        if (l >= r) {
            return;
        }
        //递归 从大区间->到小区间

        int lt = l-1, i = lt + 1, gt = r + 1;
        while (i < gt) {
            if (arrs[i]==0) {
                lt++;
                swap(arrs, lt, i);
                i++;
            } else if (arrs[i]==2) {
                gt--;
                swap(arrs, i, gt);
            } else {
                i++;
            }
        }
    }
    private static <E extends Comparable<E>> void swap(int[] arr, int i, int j) {
        int t = arr[i];
        arr[i] = arr[j];
        arr[j] = t;
    }
    public static void main(String[] args){
        int[] ints={2,2};
        sortColors(ints);
        System.out.println(Arrays.toString(ints));
    }
}
